#include "playlistmodel.h"

#include <qfileinfo.h>
#include <qdebug.h>

PlayListModel::PlayListModel(QObject *parent)
	: QAbstractListModel(parent)
{}

PlayListModel::~PlayListModel()
{}


int PlayListModel::rowCount(const QModelIndex& parent) const
{
    return (m_playlist && !parent.isValid()) ? m_playlist->mediaCount() : 0;
}

int PlayListModel::columnCount(const QModelIndex& parent) const
{
    return !parent.isValid() ? static_cast<int>(Column::ColumnCount) : 0;
}

QModelIndex PlayListModel::index(int row, int column, const QModelIndex& parent) const
{
    return m_playlist && !parent.isValid()
        && row >= 0 && row < m_playlist->mediaCount()
        && column >= 0 && column < static_cast<int>(Column::ColumnCount)
        ? createIndex(row, column)
        : QModelIndex();
}

QModelIndex PlayListModel::parent(const QModelIndex& child) const
{
    Q_UNUSED(child);

    return QModelIndex();
}

QVariant PlayListModel::data(const QModelIndex& index, int role) const
{
    //qDebug() << "call";
    if (index.isValid() && role == Qt::DisplayRole) {
        QVariant value = m_data[index];
        if (!value.isValid() && index.column() == static_cast<int>(Column::Title)) {
            QUrl location = m_playlist->media(index.row()).canonicalUrl();
            return QFileInfo(location.path()).fileName();
        }
        return value;
    }
    return QVariant();
}

QMediaPlaylist* PlayListModel::playlist() const
{
    return m_playlist;
}

void PlayListModel::setPlaylist(QMediaPlaylist* playlist)
{
    if (m_playlist) {
        disconnect(m_playlist, &QMediaPlaylist::mediaAboutToBeInserted, this, &PlayListModel::beginInsertItems);
        disconnect(m_playlist, &QMediaPlaylist::mediaInserted, this, &PlayListModel::endInsertItems);
        //disconnect(m_playlist, SIGNAL(mediaAboutToBeRemoved(int, int)), this, SLOT(beginRemoveItems(int, int)));
        //disconnect(m_playlist, SIGNAL(mediaRemoved(int, int)), this, SLOT(endRemoveItems()));
        disconnect(m_playlist, SIGNAL(mediaChanged(int, int)), this, SLOT(changeItems(int, int)));
    }

    beginResetModel();
    m_playlist = playlist;

    if (m_playlist) {
        connect(m_playlist, &QMediaPlaylist::mediaAboutToBeInserted, this, &PlayListModel::beginInsertItems);
        connect(m_playlist, &QMediaPlaylist::mediaInserted, this, &PlayListModel::endInsertItems);
        connect(m_playlist, &QMediaPlaylist::mediaAboutToBeRemoved, this, &PlayListModel::beginRemoveItems);
        connect(m_playlist, &QMediaPlaylist::mediaRemoved, this, &PlayListModel::endRemoveItems);
        connect(m_playlist, SIGNAL(mediaChanged(int, int)), this, SLOT(changeItems(int, int)));
    }

    endResetModel();
}

bool PlayListModel::setData(const QModelIndex& index, const QVariant& value, int role)
{
    Q_UNUSED(role);
    m_data[index] = value;
    emit dataChanged(index, index);
    return true;
}

void PlayListModel::beginInsertItems(int start, int end)
{
    m_data.clear();
    beginInsertRows(QModelIndex(), start, end);
}

void PlayListModel::endInsertItems()
{
    endInsertRows();
}

void PlayListModel::beginRemoveItems(int start, int end)
{
    m_data.clear();
    beginRemoveRows(QModelIndex(), start, end);
}

void PlayListModel::endRemoveItems()
{
    endRemoveRows();
}

void PlayListModel::changeItems(int start, int end)
{
    m_data.clear();
    emit dataChanged(index(start, 0), index(end, static_cast<int>(Column::ColumnCount)));
}